bilinear_shapes = [
    (32, 1024, 1024, 32),
    (64, 512, 1024, 32),
    (128, 256, 512, 64),
    (256, 128, 256, 128),
    (512, 64, 128, 256),
]